package java8;

import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import java.util.stream.Stream;

/**
 * @author lzy
 * @version v1.0
 * Create DateTime: 2018/7/19 11:35
 */
public class TestStreamAPI2 {

    List<Employee> employees = Arrays.asList(
            new Employee("张三", 18, 9999.99),
            new Employee("李四", 38, 5555.555),
            new Employee("王五", 50, 6666.66),
            new Employee("赵柳", 16, 3333.33),
            new Employee("田七", 8, 7777.77),
            new Employee("田七", 8, 7777.77),
            new Employee("田七", 8, 7777.77));

    private static Stream<Character> filterCharacter(String str) {
        List<Character> list = new ArrayList<>();
        for (Character c : str.toCharArray()) {
            list.add(c);
        }
        return list.stream();
    }

    /**
     * 中间操作
     * 筛选与切片：
     * filter--接收Lambda，从流中排除某些元素
     * limit--截断流，使其元素不超过给定数量
     * skip(n)--跳过元素，返回一个扔掉了前n个元素的流，若流中元素不足n个，则返回一个空流，与limit(n)互补
     * distinct--筛选，通过流所生成元素的hashCode()和equals()去除重复元素
     */

    // 内部迭代：迭代操作由Stream API完成
    @Test
    public void test1() {
        // 中间操作，不会执行任何操作
        Stream<Employee> s = employees.stream()
                .filter(x -> x.getAge() > 35);
        // 终止操作，一次性执行全部内容，称为"惰性求值"
        s.forEach(System.out::println);
    }

    // 外部迭代
    @Test
    public void test2() {
        Iterator<Employee> iterator = employees.iterator();
        while (iterator.hasNext()) {
            Employee e = iterator.next();
            if (e.getAge() > 35) {
                System.out.println(e);
            }
        }
    }

    @Test
    public void test3() {
        employees.stream()
                .filter(e -> {
                    System.out.println("短路");
                    return e.getSalary() > 5000;
                })
                .limit(2)
                .forEach(System.out::println);
    }

    @Test
    public void test4() {
        employees.stream()
                .filter(x -> x.getSalary() > 5000)
                .skip(2)
                .distinct()
                .forEach(System.out::println);
    }

    /**
     * 映射
     * map--接收Lambda，将元素转换为其他形式或提取信息，接收一个函数作为参数，该函数将会应用到每个元素上，并将其映射为一个新的元素
     * flatMap--接收一个函数作为参数，将流中的每个值都转换成另一个流，然后把所有流连接成一个流
     */
    @Test
    public void test5() {
        List<String> list = Arrays.asList("aaa", "bbb", "ccc", "ddd");
        list.stream().map(String::toUpperCase).forEach(System.out::println);
        System.out.println("----------------------");

        employees.stream()
                .map(Employee::getName)
                .distinct()
                .forEach(System.out::println);
    }

    @Test
    public void test6() {
        List<String> list = Arrays.asList("aaa", "bbb", "ccc", "ddd");
        /*
        Stream<Stream<Character>> stream = list.stream().map(TestStreamAPI2::filterCharacter);
        stream.forEach(sm -> sm.forEach(System.out::println));*/

        Stream<Character> stream1 = list.stream()
                .flatMap(TestStreamAPI2::filterCharacter);
        stream1.forEach(System.out::println);
    }

    /**
     * 排序
     * sorted()--自然排序(Comparable)
     * sorted(Comparator com)--定制排序
     */
    @Test
    public void test7() {
        List<String> list = Arrays.asList("aad", "dbd", "dfa", "adf", "dfdsf");
        list.stream()
                .sorted().forEach(System.out::println);

        System.out.println("--------------");

        employees.stream()
                .sorted((e1, e2) -> {
                    if (e1.getAge() == e2.getAge()) {
                        return e1.getName().compareTo(e2.getName());
                    } else {
                        return Integer.compare(e1.getAge(), e2.getAge());
                    }
                }).forEach(System.out::println);
    }
}
